how to plot cos x curve in python